from dotenv import load_dotenv
load_dotenv()
from langchain.prompts.chat import ChatPromptTemplate

#template = "你是一个数学家， 你可以计算任何算式”
template = "你是一个翻译学家，擅长将{input_language}语言翻译成{output_language}语言。"
human_template = "{text}"


chat_prompt = ChatPromptTemplate.from_messages([
    ("system", template),
    ("human", human_template),
])

#print(chat_prompt)
#exit()
#导入ChatOpenAi接口
from langchain_openai import ChatOpenAI

#创建模型实例
model = ChatOpenAI(tiktoken_model_name='gpt-4o')
#输入提示
#messages = chat_prompt.format_messages(text="我今年18岁，我的舅舅今年38岁，我的爷爷今年72岁，我和舅舅一共多少岁了？”）
#print(messages)
messages = chat_prompt.format_messages( input_language="英文", output_language = "中文", text = "I love large language Model.")
print(messages)
#得到模型输出
output = model.invoke(messages)
#打印输出内容
print(output.content)

